# -*- coding: utf-8 -*-
import pymysql

# 打开数据库连接
connection = pymysql.connect(host = '',
                            user = '',
                            password = '',
                            db = '',
                            charset = 'utf8')

# 使用 cursor() 方法创建一个游标对象 cursor
cursor = connection.cursor()
# 使用 execute()  方法执行 SQL 查询 
cursor.execute("SELECT id,title from lunwen")
title_list = cursor.fetchall()

# 遍历标题列表，对数据库进行查询，然后对作者进行关系的连接
for item in title_list:
    title_id = item[0]
    title_name = item[1]

    # 查询当前论文下的所有作者
    cursor.execute("select a_name from author_info where a_title='%s'"%title_name)
    author_list = cursor.fetchall()

    # 对所有作者进行俩俩之间的连接
    # 注意：如果论文的作者只有一个人，那么不做连接
    num_author = len(author_list)
    conn_list = []
    for i in range(num_author):
        for j in range(num_author-i-1):
            conn_list.append((author_list[i],author_list[i+j+1],title_name))
            conn_list.append((author_list[i+j+1],author_list[i],title_name))
    
    # 插入数据到数据库中
    sql = "INSERT into author_conn (name_1,name_2,title) values (%s,%s,%s)"
    cursor.executemany(sql,conn_list)
    print(title_id)
    connection.commit()

connection.close()
